package com.hub.realtime.common.utils;

import java.util.Map;

public class HadoopEnvUtil {

    public static void setNewEnv(String yarnConfDir, String hadoopConfDir) {

        Map<String, String> envMap = System.getenv();
        if (envMap != null && !envMap.isEmpty()) {
            if (envMap.containsKey("YARN_CONF_DIR")) {
                envMap.remove("YARN_CONF_DIR");
            } else {
                envMap.put("YARN_CONF_DIR", yarnConfDir);
            }

            if (envMap.containsKey("HADOOP_CONF_DIR")) {
                envMap.remove("HADOOP_CONF_DIR");
            } else {
                envMap.put("HADOOP_CONF_DIR", hadoopConfDir);
            }
        }


    }

}
